package com.example.lotterysystem.common.utils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import java.util.Collection;
import java.util.concurrent.TimeUnit;

@Configuration
public class RedisUtil {

    private static final Logger logger = LoggerFactory.getLogger(RedisUtil.class);

    /**
     * RedisTemplate: 先将被存储的数据转换成 字节数组(不可读), 再存储到redis中, 读取的时候按字节数组读取
     * StringRedisTemplate: 直接存取的就是 string(可读)
     */
    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 设置值
     * @param key
     * @param value
     * @return
     */
    public boolean set(String key, String value) {
        try {
            stringRedisTemplate.opsForValue().set(key, value);
        }catch (Exception e) {
            logger.error("RedisUtil error, set({}, {})", key, value, e);
            return false;
        }
        return true;
    }

    /**
     * 设置值(设置过期时间)
     * @param key
     * @param value
     * @param time (秒)
     * @return
     */
    public boolean set(String key, String value, Long time) {
        try {
            stringRedisTemplate.opsForValue().set(key, value,time, TimeUnit.SECONDS);
        }catch (Exception e) {
            logger.error("RedisUtil error, set({},{}, {})", key, value,time, e);
            return false;
        }
        return true;
    }

    public String get(String key) {
        try {
            return StringUtils.hasText(key)
                    ? stringRedisTemplate.opsForValue().get(key)
                    : null;
        }catch (Exception e) {
            logger.error("RedisUtil error, get({})", key, e);
            return null;
        }
    }

    /**
     * 删除值
     */
    public boolean del(String... key) {
        try {
            if(null != key && key.length > 0) {
                if(key.length == 1) {
                    stringRedisTemplate.delete(key[0]);
                }else {
                    stringRedisTemplate.delete(
                            (Collection<String>) CollectionUtils.arrayToList(key)
                    );
                }
            }
        }catch (Exception e) {
            logger.error("RedisUtil error, del({})", key, e);
            return false;
        }
        return true;
    }

    /**
     * 判断key是否存在
     * @param key
     * @return
     */
    public boolean hasKey(String key) {
       try {
           return StringUtils.hasText(key)
                   ? stringRedisTemplate.hasKey(key)
                   : false;
       }catch (Exception e) {
           logger.error("RedisUtil error, hasKey({})", key, e);
           return false;
       }
    }

}
